#!/usr/bin/python
#coding:utf-8
'''
excel工具类
'''

import xlrd
import xlwt

def read(fileName="./ts_sence_feature.xlsx",sheetname="host"):
    '''
    读取exel
    fileName:excel文件名
    sheetname:分栏名
    '''

    book = xlrd.open_workbook(fileName)#打开一个excel
    sheet = book.sheet_by_name(sheetname) #book.sheet_by_index(0)#根据顺序获取sheet
    
    print(sheet.cell(0,0).value) #指定行列
    print(sheet.ncols)#获取excel里面有多少列
    print(sheet.nrows)#获取excel里面有多少行
    print(sheet.row_values(0))#获取第一行
    print(sheet.col_values(1))#取第一列的数据
    print(sheet.get_rows())#获取所有行数据

    for row in range(sheet.nrows):  #获取行数
        if row==0:
            continue
        url=sheet.cell(row,0).value
        print(url)



def write(fileName="./result.xlsx",sheetname="result",data=[[]]):
    '''
    写入内容
    fileName:文件名
    sheetname:分栏名
    data:二维数据，第一维为整行数据，第二维是每行的单元数据
    '''
    book = xlwt.Workbook()#新建一个excel
    sheet = book.add_sheet(sheetname)#添加一个sheet页

    row=0 #控制行
    for one in data:
        col=0 #控制列
        for context in one:
            sheet.write(row,col,context)
            col+1
        row+1
    
    book.save(fileName)


'''
新版本xlsx文件读取 
'''
import openpyxl
 
# 打开xlsx文件
workbook = openpyxl.load_workbook(file)
sheet=workbook["资产填报"]

# 获取所有的工作表
sheets = workbook.sheetnames
# 遍历工作表
for sheet_name in sheets:
    # 根据工作表名获取工作表对象
    sheet = workbook[sheet_name]
    
    # 遍历工作表的所有行
    for row in sheet.iter_rows(values_only=True):
        # 对每一行进行处理
        print(row)  # 打印每一行的数据


